/*
 * This file is part of aion-unique <aion-unique.com>.
 *
 *  aion-unique is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  aion-unique is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with aion-unique.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.aionengine.gameserver.dataholders;

import com.aionengine.gameserver.model.templates.goods.GoodsList;
import gnu.trove.map.hash.TIntObjectHashMap;

import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import java.util.List;

/**
 * @author ATracer
 */
@XmlRootElement(name = "goodslists")
@XmlAccessorType(XmlAccessType.FIELD)
public class GoodsListData {

    @XmlElement(required = true)
    protected List<GoodsList> list;

    @XmlElement(name = "in_list")
    protected List<GoodsList> inList;

    @XmlElement(name = "purchase_list")
    protected List<GoodsList> purchaseList;

    /**
     * A map containing all goodslist templates
     */
    private TIntObjectHashMap<GoodsList> goodsListData;
    private TIntObjectHashMap<GoodsList> goodsInListData;
    private TIntObjectHashMap<GoodsList> goodsPurchaseListData;

    void afterUnmarshal(Unmarshaller u, Object parent) {
        goodsListData = new TIntObjectHashMap<GoodsList>();
        for (GoodsList it : list) {
            goodsListData.put(it.getId(), it);
        }
        goodsInListData = new TIntObjectHashMap<GoodsList>();
        for (GoodsList it : inList) {
            goodsInListData.put(it.getId(), it);
        }
        goodsPurchaseListData = new TIntObjectHashMap<GoodsList>();
        for (GoodsList it : purchaseList) {
            goodsPurchaseListData.put(it.getId(), it);
        }
        list = null;
        inList = null;
        purchaseList = null;
    }

    public GoodsList getGoodsListById(int id) {
        return goodsListData.get(id);
    }

    public GoodsList getGoodsInListById(int id) {
        return goodsInListData.get(id);
    }

    public GoodsList getGoodsPurchaseListById(int id) {
        return goodsPurchaseListData.get(id);
    }

    /**
     * @return goodListData.size()
     */
    public int size() {
        return goodsListData.size() + goodsInListData.size() + goodsPurchaseListData.size();
    }
}
